tensor([[0.4700, 0.4700, 0.4700, 0.4700, 0.4700],
        [0.5300, 0.5300, 0.5300, 0.5300, 0.5300]], grad_fn=<RoundBackward1>)